/*-----------------------------------------------------------------------*/
/* Low level disk I/O module SKELETON for FatFs     (C)ChaN, 2019        */
/*-----------------------------------------------------------------------*/
/* If a working storage control module is available, it should be        */
/* attached to the FatFs via a glue function rather than modifying it.   */
/* This is an example of glue functions to attach various exsisting      */
/* storage control modules to the FatFs module with a defined API.       */
/*-----------------------------------------------------------------------*/

#include "ff.h"			/* Obtains integer types */
#include "diskio.h"		/* Declarations of disk functions */
#include "wm_sdio.h"

/* Definitions of physical drive number for each drive */
#define DEV_MMC		0	/* Example: Map MMC/SD card to physical drive 1 */
#define DEV_RAM		1	/* Example: Map Ramdisk to physical drive 0 */
#define DEV_USB		2	/* Example: Map USB MSD to physical drive 2 */

extern SD_HandleTypeDef hsd;

static int RAM_disk_status(void)
{
	return 0;
}

static int MMC_disk_status(void)
{
	int state = STA_NOINIT;
	
	if (HAL_SD_GetCardState(&hsd) == HAL_SD_CARD_TRANSFER)
	{
		state &= ~STA_NOINIT;
	}
	return state;
}

static int USB_disk_status(void)
{
	return 0;
}

static int RAM_disk_initialize(void)
{
	return 0;
}

static int MMC_disk_initialize(void)
{
	int state = STA_NOINIT;
	
	if (HAL_SD_Init(&hsd) == HAL_OK)
	{
		state &= ~STA_NOINIT;
	}
	return state;
}

static int USB_disk_initialize(void)
{
	return 0;
}

static int RAM_disk_read(BYTE *buff, LBA_t sector, UINT count)
{
	return 0;
}

static int MMC_disk_read(BYTE *buff, LBA_t sector, UINT count)
{
	DRESULT res = RES_ERROR;
	uint32_t timeout = 100000;
	uint32_t buflen = BLOCKSIZE * count;
	BYTE *rdbuff = buff;
	
	if (((uint32_t)buff) & 0x00000003)	// non aligned 4byte
	{
		rdbuff = malloc(buflen);
		if (rdbuff == NULL)
		{
			return res;
		}
	}
	
	if (HAL_SD_ReadBlocks(&hsd, (uint8_t *)rdbuff, (uint32_t)sector, count, 100000000) == HAL_OK)
	{
		while(HAL_SD_GetCardState(&hsd) != HAL_SD_CARD_TRANSFER)
		{
			if (timeout-- == 0)
			{
				break;
			}
		}
		res = RES_OK;
	}
	
	if (rdbuff != buff)
	{
		if (res == RES_OK)
		{
			memcpy(buff, rdbuff, buflen);
		}
		free(rdbuff);
	}
	
	return res;
}

static int USB_disk_read(BYTE *buff, LBA_t sector, UINT count)
{
	return 0;
}

static int RAM_disk_write(const BYTE *buff, LBA_t sector, UINT count)
{
	return 0;
}

static int MMC_disk_write(const BYTE *buff, LBA_t sector, UINT count)
{
	DRESULT res = RES_ERROR;
	uint32_t timeout = 100000;
	uint32_t buflen = BLOCKSIZE * count;
	BYTE *wrbuff = (BYTE *)buff;
	
	if (((uint32_t)buff) & 0x00000003)	// non aligned 4byte
	{
		wrbuff = malloc(buflen);
		if (wrbuff == NULL)
		{
			return res;
		}
		memcpy(wrbuff, buff, buflen);
	}
	
	if (HAL_SD_WriteBlocks(&hsd, (uint8_t *)wrbuff, (uint32_t)sector, count, 100000000) == HAL_OK)
	{
		while(HAL_SD_GetCardState(&hsd) != HAL_SD_CARD_TRANSFER)
		{
			if (timeout-- == 0)
			{
				break;
			}
		}
		res = RES_OK;
	}
	
	if (wrbuff != buff)
	{
		free(wrbuff);
	}
	return res;
}

static int USB_disk_write(const BYTE *buff, LBA_t sector, UINT count)
{
	return 0;
}

/*-----------------------------------------------------------------------*/
/* Get Drive Status                                                      */
/*-----------------------------------------------------------------------*/

DSTATUS disk_status (
	BYTE pdrv		/* Physical drive nmuber to identify the drive */
)
{
	DSTATUS stat = STA_NOINIT;
	int result;

	switch (pdrv) {
	case DEV_RAM :
		result = RAM_disk_status();

		// translate the reslut code here

		return stat;

	case DEV_MMC :
		result = MMC_disk_status();

		// translate the reslut code here
		stat = result;
		return stat;

	case DEV_USB :
		result = USB_disk_status();

		// translate the reslut code here

		return stat;
	}
	return STA_NOINIT;
}



/*-----------------------------------------------------------------------*/
/* Inidialize a Drive                                                    */
/*-----------------------------------------------------------------------*/

DSTATUS disk_initialize (
	BYTE pdrv				/* Physical drive nmuber to identify the drive */
)
{
	DSTATUS stat = STA_NOINIT;
	int result;

	switch (pdrv) {
	case DEV_RAM :
		result = RAM_disk_initialize();

		// translate the reslut code here

		return stat;

	case DEV_MMC :
		result = MMC_disk_initialize();

		// translate the reslut code here
		stat = result;
		return stat;

	case DEV_USB :
		result = USB_disk_initialize();

		// translate the reslut code here

		return stat;
	}
	return STA_NOINIT;
}



/*-----------------------------------------------------------------------*/
/* Read Sector(s)                                                        */
/*-----------------------------------------------------------------------*/

DRESULT disk_read (
	BYTE pdrv,		/* Physical drive nmuber to identify the drive */
	BYTE *buff,		/* Data buffer to store read data */
	LBA_t sector,	/* Start sector in LBA */
	UINT count		/* Number of sectors to read */
)
{
	DRESULT res = RES_ERROR;
	int result;

	switch (pdrv) {
	case DEV_RAM :
		// translate the arguments here

		result = RAM_disk_read(buff, sector, count);

		// translate the reslut code here

		return res;

	case DEV_MMC :
		// translate the arguments here

		result = MMC_disk_read(buff, sector, count);

		// translate the reslut code here
		res = result;
		return res;

	case DEV_USB :
		// translate the arguments here

		result = USB_disk_read(buff, sector, count);

		// translate the reslut code here

		return res;
	}

	return RES_PARERR;
}



/*-----------------------------------------------------------------------*/
/* Write Sector(s)                                                       */
/*-----------------------------------------------------------------------*/

#if FF_FS_READONLY == 0

DRESULT disk_write (
	BYTE pdrv,			/* Physical drive nmuber to identify the drive */
	const BYTE *buff,	/* Data to be written */
	LBA_t sector,		/* Start sector in LBA */
	UINT count			/* Number of sectors to write */
)
{
	DRESULT res = RES_ERROR;;
	int result;

	switch (pdrv) {
	case DEV_RAM :
		// translate the arguments here

		result = RAM_disk_write(buff, sector, count);

		// translate the reslut code here

		return res;

	case DEV_MMC :
		// translate the arguments here

		result = MMC_disk_write(buff, sector, count);

		// translate the reslut code here
		res = result;
		return res;

	case DEV_USB :
		// translate the arguments here

		result = USB_disk_write(buff, sector, count);

		// translate the reslut code here

		return res;
	}

	return RES_PARERR;
}

#endif


/*-----------------------------------------------------------------------*/
/* Miscellaneous Functions                                               */
/*-----------------------------------------------------------------------*/

DRESULT disk_ioctl (
	BYTE pdrv,		/* Physical drive nmuber (0..) */
	BYTE cmd,		/* Control code */
	void *buff		/* Buffer to send/receive control data */
)
{
	DRESULT res = RES_ERROR;
	HAL_SD_CardInfoTypeDef CardInfo;

	switch (pdrv) {
	case DEV_RAM :

		// Process of the command for the RAM drive

		return res;

	case DEV_MMC :

		// Process of the command for the MMC/SD card
		switch(cmd)
		{
			case CTRL_SYNC:
				res = RES_OK;
				break;
				
			case GET_SECTOR_COUNT:
				HAL_SD_GetCardInfo(&hsd, &CardInfo);
				*(DWORD*)buff = CardInfo.LogBlockNbr;
				res = RES_OK;
				break;
				
			case GET_SECTOR_SIZE:
				HAL_SD_GetCardInfo(&hsd, &CardInfo);
				*(DWORD*)buff = CardInfo.LogBlockSize;
				res = RES_OK;
				break;
				
			case GET_BLOCK_SIZE:
				HAL_SD_GetCardInfo(&hsd, &CardInfo);
				*(DWORD*)buff = CardInfo.LogBlockSize;
				res = RES_OK;
				break;
				
			default:
				res = RES_PARERR;
		}

		return res;

	case DEV_USB :

		// Process of the command the USB drive

		return res;
	}

	return RES_PARERR;
}

